import smtplib
from email.mime.text import MIMEText
from email.header import Header
import random


def send(user_email, code):
    print('发送。。。')
    # 第三方 SMTP 服务
    mail_host = "smtp.qq.com"  # 设置服务器
    mail_user = "1648639361@qq.com"  # 用户名
    mail_pass = "hswzkfdojiipbcbf"  # 口令

    sender = '1648639361@qq.com'
    receivers = [user_email]  # 接收邮件，可设置为你的QQ邮箱或者其他邮箱 'mjiaqi528@gmail.com'

    email_content = '亲爱的用户:\n \t您的验证码为:' + str(code)
    message = MIMEText(email_content, 'plain', 'utf-8')
    message['From'] = Header("Audio听书", 'utf-8')
    message['To'] = Header("尊敬的用户", 'utf-8')

    subject = 'Audio听书注册验证码'
    message['Subject'] = Header(subject, 'utf-8')

    try:
        smtpObj = smtplib.SMTP()
        smtpObj.connect(mail_host, 25)  # 25 为 SMTP 端口号
        smtpObj.login(mail_user, mail_pass)
        smtpObj.sendmail(sender, receivers, message.as_string())
        print("邮件发送成功")
    except smtplib.SMTPException:
        print("Error: 无法发送邮件")


# 生成验证码
def getCode():
    code = random.choice(range(100000, 1000000))
    return code
